<?php
/**
 * Created by PhpStorm.
 * User: admin@zhangp.cn
 * Date: 2018/4/5
 * Time: 14:37
 */

/**
 * 功能: WebSocket PHP客户端例子
 * Class WebSocketClient
 */
class WebSocketClient
{
    private $_host;
    private $_port;
    private $_path;
    private $_origin;
    private $_Socket = null;
    private $_connected = false;

    public function __construct() { }

    public function __destruct()
    {
        $this->disconnect();
    }


    /**
     * 功能:发送数据
     * @param $msgType
     * @param $data
     * @return bool
     */
    public function sendData($msgType, $data, $callback = null)
    {
        if($this->_connected === false)
        {
            trigger_error("Not connected", E_USER_WARNING);
            return false;
        }

        if(empty($data)){
            trigger_error("Data is empty", E_USER_WARNING);
            return false;
        }

       // $data['sign'] = $this->getSysSign($data);

        $data = json_encode($data);

        $sendData = [
            'MsgType'   =>  $msgType,
            'Data'      =>  $data,
            'Nonce'     =>  $this->getNonceStr(10, 6)
        ];


        $sendData['Sign'] = $this->getSign($sendData);

        $sendData = json_encode($sendData);

        $sendData = $this->_hybi10Encode($sendData);

        $res = socket_write($this->_Socket, $sendData, strlen($sendData));

        if($res === 0 || $res === false)
        {
            return false;
        }
        set_time_limit(5);
        $status = false;
        while (true){
            $bytes = socket_recv($this->_Socket, $buffer, 2048, 0);
            $response = $this->_hybi10Decode($buffer);
            $responseData = json_decode($response, true);
            //var_dump($responseData);
           break;

//            $bytes = socket_recv($this->_Socket, $buffer, 2048, 0);
//            $response = $this->_hybi10Decode($buffer);
////            var_dump($response);
//            $responseData = json_decode($response, true);
//            break;
//            if($callback){
//                call_user_func_array($callback,array(&$status, $responseData));
//            } else {
//                $status = true;
//            }
//            if($status) break;
        }

    }




    /**
     * 功能:创建socket连接
     * @param $host
     * @param $port
     * @param $path
     * @param bool $origin
     * @return bool
     */
    public function connect($host, $port, $path, $origin = false)
    {
        $this->_host = $host;
        $this->_port = $port;
        $this->_path = $path;
        $this->_origin = $origin;

        $key = base64_encode($this->_generateRandomString(16, false, true));
        $header = "GET " . $path . " HTTP/1.1\r\n";
        $header.= "Host: ".$host.":".$port. "\r\n";
        $header.= "Upgrade: websocket\r\n";
        $header.= "Connection: Upgrade\r\n";
        //$header.= "Sec-WebSocket-Extensions: permessage-deflate; client_max_window_bits\r\n";
        $header.= "Sec-WebSocket-Key: " . $key . "\r\n";

        if($origin !== false)
        {
            $header.= "Sec-WebSocket-Origin: " . $origin . "\r\n";
        }
        $header.= "Sec-WebSocket-Version: 13\r\n\r\n";

//        $this->_Socket = fsockopen($host, $port, $errno, $errstr, 2);

        $this->_Socket = socket_create(AF_INET,SOCK_STREAM,SOL_TCP);
        socket_connect($this->_Socket, $host, $port);
//        socket_set_timeout($this->_Socket, 2, 10000);
        //socket_write($this->_Socket, $header);
        $res = socket_write($this->_Socket, $header, strlen($header));
        if( $res === false ){
            echo "fwrite false \n";
        }
//        $response = @fread($this->_Socket, 1500);
        $response = socket_read($this->_Socket, 2048);
        preg_match('#Sec-WebSocket-Accept:\S*(.*)$#mU', $response, $matches);
        if ($matches) {
            $keyAccept = trim($matches[1]);
            $expectedResonse = base64_encode(pack('H*', sha1($key . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11')));
            $this->_connected = ($keyAccept === $expectedResonse) ? true : false;
        }
        return $this->_connected;
    }

    public function checkConnection()
    {
        $this->_connected = false;

        // send ping:
        $data = 'ping?';
        @fwrite($this->_Socket, $this->_hybi10Encode($data, 'ping', true));
        $response = @fread($this->_Socket, 300);
        if(empty($response))
        {
            return false;
        }
        $response = $this->_hybi10Decode($response);
        if(!is_array($response))
        {
            return false;
        }
        if(!isset($response['type']) || $response['type'] !== 'pong')
        {
            return false;
        }
        $this->_connected = true;
        return true;
    }

    /**
     * 功能: 返回随机字符串
     * @param int $length
     * @param int $type
     * @return string
     */
    protected function getNonceStr($length = 6, $type = 0){
        switch ((int)$type){
            case 0:
                $str = '0123456789';
                break;
            case 1:
                $str = 'abcdefghijklmnopqrstuvwxyz';
                break;
            case 2:
                $str = 'ABCDEFGHIJKLMNOPQRSTUVWXYZ';
                break;
            case 3:
                $str = 'abcdefghijklmnopqrstuvwxyz0123456789';
                break;
            case 4:
                $str = 'ABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789';
                break;
            case 5:
                $str = 'abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ';
                break;
            case 6:
                $str = 'abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789';
                break;
            case 7:
                $str = 'abcdefghijkmnprstuvwxyABCDEFGHJKLMNPQRSTUVWXY345678'; //去掉了易混淆字符
                break;
            default:
                $str = 'abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789';
        }

        if($length == 0){
            return ''; //要求随机字符串长度为0;
        }

        if($length < 0){
            $length = abs($length); //如果为负数取绝对值;
        }

        $length = ceil($length); // 对长度进行向上取整,防止传入小数;

        $nonce = '';

        for(; $length > 0; $length--){
            $nonce.=substr($str, rand(0, strlen($str) - 1), 1);
        }

        return $nonce;
    }


    /**
     * 功能:对数组进行深度排序
     * @param $data
     * @return mixed
     */
    protected function deepSortArray($data){
        ksort($data, SORT_STRING);
        foreach ($data as $k=>$v){
            if($k == 'sign'){
                unset($data['sign']);
            }
            if(is_array($v)){
                $data[$k] = $this->deepSortArray($v);
            }
        }
        return $data;
    }

    /**
     * 功能:将数据转换为待签名字符串
     * @param $data
     * @return string
     */
    protected function getSysSignString($data){
        $data = $this->deepSortArray($data);

        $signString = json_encode($data) . 'www.hanguosoft.com';
        return $signString;
    }

    /**
     * 功能:返回签名
     * @param $data
     * @return string
     */
    protected function getSysSign($data){
        $dataString = $this->getSysSignString($data);
        return strtoupper(md5($dataString));
    }

    /**
     * 功能:对数据进行签名
     * @param $socket
     * @param $msg
     * @return string
     */
    protected function getSign($msg){
        $signString = "MsgType=".$msg['MsgType'];
        $signString = $signString."&"."Data=".$msg['Data'];
        $signString = $signString."&"."Nonce=".$msg['Nonce'];
        $signString = $signString."&"."key=".'';
        return strtoupper(md5($signString));
    }




    public function disconnect()
    {
        $this->_connected = false;
        socket_close($this->_Socket);
//        is_resource($this->_Socket) and fclose($this->_Socket);
    }

    public function reconnect()
    {
        sleep(10);
        $this->_connected = false;
        fclose($this->_Socket);
        $this->connect($this->_host, $this->_port, $this->_path, $this->_origin);
    }

    private function _generateRandomString($length = 10, $addSpaces = true, $addNumbers = true)
    {
        $characters = 'abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ!"ยง$%&/()=[]{}';
        $useChars = array();
        // select some random chars:
        for($i = 0; $i < $length; $i++)
        {
            $useChars[] = $characters[mt_rand(0, strlen($characters)-1)];
        }
        // add spaces and numbers:
        if($addSpaces === true)
        {
            array_push($useChars, ' ', ' ', ' ', ' ', ' ', ' ');
        }
        if($addNumbers === true)
        {
            array_push($useChars, rand(0,9), rand(0,9), rand(0,9));
        }
        shuffle($useChars);
        $randomString = trim(implode('', $useChars));
        $randomString = substr($randomString, 0, $length);
        return $randomString;
    }

    private function _hybi10Encode($payload, $type = 'text', $masked = true)
    {
        $frameHead = array();
        $frame = '';
        $payloadLength = strlen($payload);

        switch($type)
        {
            case 'text':
                // first byte indicates FIN, Text-Frame (10000001):
                $frameHead[0] = 129;
                break;

            case 'close':
                // first byte indicates FIN, Close Frame(10001000):
                $frameHead[0] = 136;
                break;

            case 'ping':
                // first byte indicates FIN, Ping frame (10001001):
                $frameHead[0] = 137;
                break;

            case 'pong':
                // first byte indicates FIN, Pong frame (10001010):
                $frameHead[0] = 138;
                break;
        }

        // set mask and payload length (using 1, 3 or 9 bytes)
        if($payloadLength > 65535)
        {
            $payloadLengthBin = str_split(sprintf('%064b', $payloadLength), 8);
            $frameHead[1] = ($masked === true) ? 255 : 127;
            for($i = 0; $i < 8; $i++)
            {
                $frameHead[$i+2] = bindec($payloadLengthBin[$i]);
            }
            // most significant bit MUST be 0 (close connection if frame too big)
            if($frameHead[2] > 127)
            {
                $this->close(1004);
                return false;
            }
        }
        elseif($payloadLength > 125)
        {
            $payloadLengthBin = str_split(sprintf('%016b', $payloadLength), 8);
            $frameHead[1] = ($masked === true) ? 254 : 126;
            $frameHead[2] = bindec($payloadLengthBin[0]);
            $frameHead[3] = bindec($payloadLengthBin[1]);
        }
        else
        {
            $frameHead[1] = ($masked === true) ? $payloadLength + 128 : $payloadLength;
        }

        // convert frame-head to string:
        foreach(array_keys($frameHead) as $i)
        {
            $frameHead[$i] = chr($frameHead[$i]);
        }
        if($masked === true)
        {
            // generate a random mask:
            $mask = array();
            for($i = 0; $i < 4; $i++)
            {
                $mask[$i] = chr(rand(0, 255));
            }

            $frameHead = array_merge($frameHead, $mask);
        }
        $frame = implode('', $frameHead);

        // append payload to frame:
        $framePayload = array();
        for($i = 0; $i < $payloadLength; $i++)
        {
            $frame .= ($masked === true) ? $payload[$i] ^ $mask[$i % 4] : $payload[$i];
        }

        return $frame;
    }

    private function _hybi10Decode($data)
    {
        $payloadLength = '';
        $mask = '';
        $unmaskedPayload = '';
        $decodedData = array();

        // estimate frame type:
        $firstByteBinary = sprintf('%08b', ord($data[0]));
        $secondByteBinary = sprintf('%08b', ord($data[1]));
        $opcode = bindec(substr($firstByteBinary, 4, 4));
        $isMasked = ($secondByteBinary[0] == '1') ? true : false;
        $payloadLength = ord($data[1]) & 127;

        switch($opcode)
        {
            // text frame:
            case 1:
                $decodedData['type'] = 'text';
                break;

            case 2:
                $decodedData['type'] = 'binary';
                break;

            // connection close frame:
            case 8:
                $decodedData['type'] = 'close';
                break;

            // ping frame:
            case 9:
                $decodedData['type'] = 'ping';
                break;

            // pong frame:
            case 10:
                $decodedData['type'] = 'pong';
                break;

            default:
                return false;
                break;
        }

        if($payloadLength === 126)
        {
            $mask = substr($data, 4, 4);
            $payloadOffset = 8;
            $dataLength = bindec(sprintf('%08b', ord($data[2])) . sprintf('%08b', ord($data[3]))) + $payloadOffset;
        }
        elseif($payloadLength === 127)
        {
            $mask = substr($data, 10, 4);
            $payloadOffset = 14;
            $tmp = '';
            for($i = 0; $i < 8; $i++)
            {
                $tmp .= sprintf('%08b', ord($data[$i+2]));
            }
            $dataLength = bindec($tmp) + $payloadOffset;
            unset($tmp);
        }
        else
        {
            $mask = substr($data, 2, 4);
            $payloadOffset = 6;
            $dataLength = $payloadLength + $payloadOffset;
        }

        if($isMasked === true)
        {
            for($i = $payloadOffset; $i < $dataLength; $i++)
            {
                $j = $i - $payloadOffset;
                if(isset($data[$i]))
                {
                    $unmaskedPayload .= $data[$i] ^ $mask[$j % 4];
                }
            }
            $decodedData = $unmaskedPayload;
        }
        else
        {
            $payloadOffset = $payloadOffset - 4;
            $decodedData = substr($data, $payloadOffset);
        }

        return $decodedData;
    }
}
/**********用例************************/
//$ws = new WebSocketClient();
//$ws->connect('localhost', '3004', '/');
//$ws->sendData('registerSysNotice', [
//    'msg'   =>  '测试通讯'
//]);

//var_dump($ws->sendData([
//    'MsgType'   =>  'test',
//    'Data'      =>  json_encode([1,2,3,4]),
//    'Sign'      =>  'aaaaaaaaaaa',
//    'Nonce'     =>  'aaaaaaa'
//]));
